package com.ruoyi.demo.jsoup.demo1;

import cn.hutool.db.Db;
import cn.hutool.db.Entity;
import com.ruoyi.demo.jsoup.demo1.domain.Content;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

import java.net.URL;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author : zhenhc
 * @date : 2022-04-23 09:15
 **/
public class JDCrawler {
    public static void main(String[] args) {
        //进行测试，调用方法，输入想要爬取数据的关键字
        JDCrawler jdCrawler = new JDCrawler();
        List<Content> list = null;
        try {
            list = jdCrawler.parseJD("sony");
        } catch (Exception e) {
            e.printStackTrace();
        }
        list.forEach(content -> {
            try {
                Db.use().insert(
                        Entity.create("jsoup_product")
                                .set("title",content.getTitle())
                                .set("image",content.getImage())
                                .set("price",content.getPrice())
                                .set("unit",content.getUnit())
                );
            } catch (SQLException e) {
                e.printStackTrace();
            }
        });
    }
    public List<Content> parseJD(String keywords) throws Exception {
        //获取请求，我们使用京东页面作为演示例子：http://search.jd.com/Search?keyword=[keywords]
        //注意：爬取数据过程需要联网
        String url = "http://search.jd.com/Search?keyword=" + keywords;

        //得到url后，解析网页，该 Document对象 就是浏览器 Document对象
        Document document = Jsoup.parse(new URL(url), 30000);
        Element element = document.getElementById("J_goodsList");
        //获取所有的li元素
        Elements elements = element.getElementsByTag("li");
        //创建ArrayList集合，将所有的查询对象添加到集合中
        ArrayList<Content> goodsList = new ArrayList<>();
        //获取li元素中的内容，这里的el就是每一个li元素
        for (Element el : elements) {
            //关于这种图片特别多的网站，所有的图片都是延时加载的
            //代码中的p-price,p-name等属性名是通过手动审查网页元素得到
            String img = "https:"+el.getElementsByTag("img").eq(0).attr("data-lazy-img");
            String price = el.getElementsByClass("p-price").eq(0).text().replaceAll("￥","");
            String title = el.getElementsByClass("p-name").eq(0).text();
            String unit = "￥";
            goodsList.add(new Content(title, img, price,unit));
        }
        return goodsList;
    }
}
